from flask_wtf import FlaskForm
from wtforms import SubmitField, RadioField
from wtforms.fields.simple import TextAreaField
from wtforms.validators import DataRequired, Length


message = '这个问题必选'

class QuestionnaireForm(FlaskForm):
    q1 = RadioField('1. 请选择您的年龄区间：', choices=[
        ('1', '6-12岁'),
        ('2', '12-18岁'),
        ('3', '18-24岁'),
        ('4', '24-30岁'),
        ('5', '30-40岁'),
        ('6', '40-50岁'),
        ('7', '50岁以上')
    ], validators=[DataRequired(message=message)])
    q2 = RadioField('2. 请问您听说过杨浦图书馆吗？',
        choices=[
            ('1', '听说过'),
            ('2', '没有听说过')
        ], validators=[DataRequired(message=message)]
    )
    q3 = RadioField('3. 请问您去杨浦图书馆的频率是？', choices=[
        ('1', '从来没有'),
        ('2', '去过，但记不清'),
        ('3', '每年一次'),
        ('4', '每半年一次'),
        ('5', '每月一次'),
        ('6', '每两周一次'),
        ('7', '每周一次')
    ], validators=[DataRequired(message=message)])
    q4 = RadioField('4. 请问您去杨浦图书馆的主要目的是？', choices=[
        ('0', '无'),
        ('1', '看书，查阅资料'),
        ('2', '做志愿者'),
        ('3', '其他'),
    ], validators=[DataRequired(message=message)])
    q5 = RadioField('5. 请问您对杨浦图书馆的历史了解多少（1）', choices=[
        ('0', '不知道'),
        ('1', '杨浦图书馆是在2000年后建立的'),
        ('2', '杨浦图书馆是在1978-1999之间建成的'),
        ('3', '杨浦图书馆是在1940-1978之间建成的'),
        ('4', '杨浦图书馆是在1930年代建成的'),
        ('5', '杨浦图书馆是在1930年之前建成的')
    ], validators=[DataRequired(message=message)])
    q6 = RadioField('6. 请问您对杨浦图书馆的历史了解多少(2)', choices=[
        ('0', '不知道'),
        ('1', '杨浦图书馆只是一个很普通的区级图书馆'),
        ('2', '杨浦图书馆曾是旧上海市图书馆，位于旧上海市政府附近')
    ], validators=[DataRequired(message=message)])
    q7 = RadioField('7. 请问您对杨浦图书馆的喜爱程度', choices=[
        ('0', '不知道'),
        ('1', '不喜欢'),
        ('2', '一般般'),
        ('3', '喜欢'),
        ('4', '非常喜欢')
    ], validators=[DataRequired(message=message)])
    q8 = RadioField('8. 请问您认为杨浦图书馆应该添加什么种类的书籍？', choices=[
        ('0', '不知道'),
        ('1', '计算机类'),
        ('2', '医药养生类'),
        ('3', '航空航天类'),
        ('4', '军事武器类'),
        ('5', '农业手工类'),
        ('6', '其他（不用填写内容）')
    ], validators=[DataRequired(message=message)])
    q9 = TextAreaField('请问您对杨浦图书馆有什么建议？（不得多于300字，可不填）',
                       validators=[Length(0, 300, message='不得多于300字')])
    submit = SubmitField('提交')
